#p18
import torch
from torch import nn
from torch.nn import Conv2d
import torchvision
from torch.utils.data import DataLoader

dataset = torchvision.datasets.CIFAR10("../data/conv2dData",train=False,
                                       transform=torchvision.transforms.ToTensor(),download=True)

dataloader = DataLoader(dataset, batch_size=64)

class Shenjing(nn.Module):
    def __init__(self):
        super(Shenjing, self).__init__()
        self.conv1 = Conv2d(in_channels=3, out_channels=6, kernel_size=3, stride=1, padding=0)

    def forward(self,x):
        x = self.conv1(x)
        return x

shenjing = Shenjing()
print(shenjing)